package com.mings.ai.dto;

import java.util.List;

public class EnhancedSearchResponse {
    private String answer; // 大模型生成的回答
    private List<SearchResult> relevantDocuments; // 相关文档
    private int totalResults;

    public EnhancedSearchResponse(String answer, List<SearchResult> relevantDocuments, int totalResults) {
        this.answer = answer;
        this.relevantDocuments = relevantDocuments;
        this.totalResults = totalResults;
    }

    // getters and setters
    public String getAnswer() { return answer; }
    public void setAnswer(String answer) { this.answer = answer; }
    public List<SearchResult> getRelevantDocuments() { return relevantDocuments; }
    public void setRelevantDocuments(List<SearchResult> relevantDocuments) { this.relevantDocuments = relevantDocuments; }
    public int getTotalResults() { return totalResults; }
    public void setTotalResults(int totalResults) { this.totalResults = totalResults; }
}